package com.shm.leetcode;

import java.util.ArrayList;
import java.util.Deque;
import java.util.LinkedList;
import java.util.List;

/**
 * 103. 二叉树的锯齿形层序遍历
 * 给定一个二叉树，返回其节点值的锯齿形层序遍历。（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）。
 *
 * 例如：
 * 给定二叉树 [3,9,20,null,null,15,7],
 *
 *     3
 *    / \
 *   9  20
 *     /  \
 *    15   7
 * 返回锯齿形层序遍历如下：
 *
 * [
 *   [3],
 *   [20,9],
 *   [15,7]
 * ]
 * @author SHM
 */
public class ZigzagLevelOrder {
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        List<List<Integer>> list = new ArrayList<>();
        if(root==null){
            return list;
        }
        Deque<TreeNode> deque = new LinkedList<>();
        deque.offer(root);
        //层数索引
        int count = 0;
        //层序遍历
        while(!deque.isEmpty()){
            int n = deque.size();
            List<Integer> temp = new ArrayList<>();
            for(int i = 0;i<n;i++){
                TreeNode node = deque.poll();
                if(count%2==0){
                    temp.add(node.val);
                }else {
                    //头插法
                    temp.add(0,node.val);
                }
                if(node.left!=null){
                    deque.offer(node.left);
                }
                if(node.right!=null){
                    deque.offer(node.right);
                }
            }
            list.add(temp);
            count++;
        }
        return list;
    }
}
